from cassandra.cluster import Cluster
from uuid import uuid4

# Подключение к кластеру Cassandra
cluster = Cluster(['127.0.0.1'])
session = cluster.connect('my_keyspace')

# Генерация и вставка тестовых данных в таблицу с простым ключом
for i in range(1, 5001):
    session.execute(
        """
        INSERT INTO simple_table (id, name, value) VALUES (%s, %s, %s)
        """,
        (uuid4(), f'Item{i}', i * 10)
    )

# Генерация и вставка тестовых данных в таблицу с составным ключом
for i in range(1, 5001):
    session.execute(
        """
        INSERT INTO composite_table (partition_key, clustering_key1, clustering_key2, non_primary_field) VALUES (%s, %s, %s, %s)
        """,
        ('Partition1', i, 'A', f'Data{i}')
    )

# Закрытие соединения
cluster.shutdown()
